package com.ljy.my_study.leetcode.超级洗衣机;

import java.util.Arrays;

/**
 * @author lijunying
 * @date 2021/9/29 10:31
 */
public class TestMain {

//    假设有 n 台超级洗衣机放在同一排上。开始的时候，每台洗衣机内可能有一定量的衣服，也可能是空的。
//
//    在每一步操作中，你可以选择任意 m (1 <= m <= n) 台洗衣机，与此同时将每台洗衣机的一件衣服送到相邻的一台洗衣机。
//
//    给定一个整数数组 machines 代表从左至右每台洗衣机中的衣物数量，请给出能让所有洗衣机中剩下的衣物的数量相等的 最少的操作步数 。如果不能使每台洗衣机中衣物的数量相等，则返回 -1 。
//
//             
//
//    示例 1：
//
//    输入：machines = [1,0,5]
//    输出：3
//    解释：
//    第一步:    1     0 <-- 5    =>    1     1     4
//    第二步:    1 <-- 1 <-- 4    =>    2     1     3
//    第三步:    2     1 <-- 3    =>    2     2     2
//    示例 2：
//
//    输入：machines = [0,3,0]
//    输出：2
//    解释：
//    第一步:    0 <-- 3     0    =>    1     2     0
//    第二步:    1     2 --> 0    =>    1     1     1
//    示例 3：
//
//    输入：machines = [0,2,0]
//    输出：-1
//    解释：
//    不可能让所有三个洗衣机同时剩下相同数量的衣物。
//             
//
//    提示：
//
//    n == machines.length
//1 <= n <= 104
//            0 <= machines[i] <= 105
//
//    来源：力扣（LeetCode）
//    链接：https://leetcode-cn.com/problems/super-washing-machines
//    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

    public static void main(String[] args) {
        System.out.println(new TestMain().findMinMoves(new int[]{4,0,0,4}));
        System.out.println(new TestMain().findMinMoves2(new int[]{4,0,0,4}));
    }

//    贪心算法
    public int findMinMoves(int[] machines) {
        int tot = Arrays.stream(machines).sum();
        int n = machines.length;
        if (tot % n != 0) {
            return -1;
        }
        int avg = tot / n;
        int ans = 0, sum = 0;
        for (int num : machines) {
            num -= avg;
            sum += num;
            ans = Math.max(ans, Math.max(Math.abs(sum), num));
        }
        return ans;
    }

    public int findMinMoves2(int[] ms) {
        int n = ms.length;
        int sum = 0;
        for (int i : ms) sum += i;
        if (sum % n != 0) return -1;
        int t = sum / n;
        int ls = 0, rs = sum;
        int ans = 0;
        for (int i = 0; i < n; i++) {
            rs -= ms[i];
            int a = Math.max(t * i - ls, 0);
            int b = Math.max((n - i - 1) * t - rs, 0);
            ans = Math.max(ans, a + b);
            ls += ms[i];
        }
        return ans;
    }
}
